import numpy as np


class Map:
    def __init__(self, features: np.ndarray) -> None:
        super().__init__()

        self._features = features

    def get(self, index=None):
        if index is not None:
            return self._features[index, :]
        return self._features

    def count(self):
        return self._features.shape[0]
